package test02collection;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Scanner;

/**
 * @author xhw
 */
public class _Arraylist {
    /*练习二：编写java程序，3个ArrayList对象，每个对象中添加一些学员的姓名。再创建HashMap对象，
    以年级名称为键，存放学生的ArrayList为值，然后从HashMap对象中获取某个班级的学员信息并输出*/

    public static void main(String[] args) {

        ArrayList arrayList1 = new ArrayList<>();
        arrayList1.add("a1");
        arrayList1.add("a2");
        arrayList1.add("a3");

        ArrayList arrayList2 = new ArrayList<>();
        arrayList2.add("b1");
        arrayList2.add("b2");
        arrayList2.add("b3");

        ArrayList arrayList3 = new ArrayList<>();
        arrayList3.add("c1");
        arrayList3.add("c2");
        arrayList3.add("c3");

        HashMap<String, ArrayList> hashMap = new HashMap<>(5);
        hashMap.put("小学", arrayList1);
        hashMap.put("中学", arrayList2);
        hashMap.put("大学", arrayList3);

        Scanner scanner = new Scanner(System.in);
        System.out.println("输入年级：");
        String s = scanner.nextLine();
        System.out.println(s + "学生列表:");
        ArrayList<String> o = hashMap.get(s);
        for (String o1 : o) {
            System.out.println(o1);

        }
    }
}
